package ru.ifmo.mlarinsky.classificator.classification;

import java.awt.Color;
import java.util.HashMap;
import java.util.Map;
import ru.ifmo.mlarinsky.classificator.io.InstrumentLoader;
/**
 * @author Mikhail Larinskiy
 */
public class SPClassification extends Classification {
    private static final Map<String, String> INDEX_TO_SECTOR = new HashMap<String, String>();
    private static final Map<String, Color> INDEX_TO_COLOR = new HashMap<String, Color>();
    static {
        INDEX_TO_SECTOR.put("XLB", "Business Materials");
        INDEX_TO_SECTOR.put("XLI", "Capital Goods");
        INDEX_TO_SECTOR.put("TTH", "Communication Services");
        INDEX_TO_SECTOR.put("XLF", "Financials");
        INDEX_TO_SECTOR.put("IYH", "Health Care");
        INDEX_TO_SECTOR.put("XLY", "Consumer Cyclicals");
        INDEX_TO_SECTOR.put("XLP", "Consumer Staples");
        INDEX_TO_SECTOR.put("XLK", "Technology");
        INDEX_TO_SECTOR.put("XLU", "Utilities");
        INDEX_TO_SECTOR.put("XLV", "Transportations");

        INDEX_TO_COLOR.put("XLB", new Color(255, 0, 0));
        INDEX_TO_COLOR.put("XLI", new Color(255, 178, 0));
        INDEX_TO_COLOR.put("TTH", new Color(255, 255, 0));
        INDEX_TO_COLOR.put("XLF", new Color(0, 255, 0));
        INDEX_TO_COLOR.put("IYH", new Color(0, 255, 255));
        INDEX_TO_COLOR.put("XLY", new Color(0, 0, 255));
        INDEX_TO_COLOR.put("XLP", new Color(255, 0, 215));
        INDEX_TO_COLOR.put("XLK", new Color(178, 255, 0));
        INDEX_TO_COLOR.put("XLU", new Color(0, 255, 178));
        INDEX_TO_COLOR.put("XLV", new Color(255, 100, 100));
    }

    public SPClassification() {
        super(InstrumentLoader.SP_INDEXES, INDEX_TO_SECTOR, INDEX_TO_COLOR);
    }
}
